list comprehension hackerrank